package net.huashitong.supervision.task.util;


import com.sirdc.modules.utils.StringUtils;
import net.huashitong.supervision.task.entity.TaskUnit;
import net.huashitong.supervision.task.filter.TaskSubFilter;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * POI解析Excel
 */
public class ExcelReaderUtil {

    /**
     * 根据fileType不同读取excel文件
     *  读取子任务表
     * @param
     * @throws IOException
     */
    public static List<List<String>> readExcel(InputStream inputStream, String name) {
        List<List<String>> lists = new ArrayList<List<String>>();

        String fileType = name.substring(name.lastIndexOf(".") + 1);
        InputStream is = null;
        try {
//            is = InputStream;
            //获取工作薄
            Workbook wb = null;
           /* if (fileType.equals("xls")) {
                wb = new HSSFWorkbook(inputStream);
            } else if (fileType.equals("xlsx")) {
                wb = new XSSFWorkbook(inputStream);
            } else {
                return null;
            }*/
            wb=WorkbookFactory.create(inputStream);


            //读取第一个工作页sheet
            Sheet sheet = wb.getSheetAt(0);
            //第一行为标题

            /**  读取excel表格的值，
             *  ************** update by xxq ********
             */
            int lastRow = sheet.getLastRowNum(); //获取最后一行
            for (int i = 0; i <= lastRow; i++) {
                Row row = sheet.getRow(i);   //获取excel 第i行
                String value0 = null;  //获取第i行 第1格的值
                try {
                    row.getCell(0).setCellType(Cell.CELL_TYPE_STRING);  //设置第i行 第1格
                    value0 = row.getCell(0).getStringCellValue();
                } catch (Exception e) {
//                    e.printStackTrace();
                    value0="";
                }

                String value1 = null;  //获取第i行 第2格的值
                try {
                    row.getCell(1).setCellType(Cell.CELL_TYPE_STRING);  //设置第i行 第2格
                    value1 = row.getCell(1).getStringCellValue();
                } catch (NullPointerException e) {
//                    e.printStackTrace();
                    value1="";
                }

                String value2 = null;  //获取第i行 第3格的值
                try {
                    row.getCell(2).setCellType(Cell.CELL_TYPE_STRING);  //设置第i行 第3格
                    value2 = row.getCell(2).getStringCellValue();
                } catch (NullPointerException e) {
//                    e.printStackTrace();
                    value2="";
                }
                String value3 = null;  //获取第i行 第4格的值
                try {
                    row.getCell(3).setCellType(Cell.CELL_TYPE_STRING);  //设置第i行 第4格
                    value3 = row.getCell(3).getStringCellValue();
                } catch (NullPointerException e) {
//                    e.printStackTrace();
                    value3="";
                }

                String value4 ="";
                try {

                    row.getCell(4).setCellType(Cell.CELL_TYPE_STRING);//社会第i列   第5列
                    value4=row.getCell(4).getStringCellValue();
                }catch (NullPointerException e)
                {
                    value4="";
                }

                ArrayList<String> list = new ArrayList<String>();
                list.add(value0);
                list.add(value1);
                list.add(value2);
                list.add(value3);
                list.add(value4);
                lists.add(list);
            }

           /* for (Row row : sheet) {
                ArrayList<String> list = new ArrayList<String>();
                for (Cell cell : row) {
                    //根据不同类型转化成字符串
                    cell.setCellType(Cell.CELL_TYPE_STRING);
                    list.add(cell.getStringCellValue());
                }

                lists.add(list);
            }*/
        } catch (IOException e) {
            e.printStackTrace();
        } catch (InvalidFormatException e) {

        } finally {
            try {
                if (inputStream != null) inputStream.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        return lists;
    }
/**
     * 根据fileType不同读取excel文件
     *  读取单位导入用户信息表
     * @param
     * @throws IOException
     */
    public static List<List<String>> readUserExcel(InputStream inputStream, String name) {
        List<List<String>> lists = new ArrayList<List<String>>();

        String fileType = name.substring(name.lastIndexOf(".") + 1);
        InputStream is = null;
        try {
//            is = InputStream;
            //获取工作薄
            Workbook wb = null;

            wb=WorkbookFactory.create(inputStream);
            //读取第一个工作页sheet
            Sheet sheet = wb.getSheetAt(0);
            //第一行为标题

            /**  读取excel表格的值，
             *  ************** update by xxq ********
             */
            int lastRow = sheet.getLastRowNum(); //获取最后一行
            for (int i = 0; i <= lastRow; i++) {
                Row row = sheet.getRow(i);   //获取excel 第i行
                String value0 = "";  //获取第i行 第1格的值
                String value1 = "";  //获取第i行 第2格的值
                String value2 = "";  //获取第i行 第3格的值
                String value3 = "";  //获取第i行 第4格的值
                Cell cell = row.getCell(0);
                Cell cell1 = row.getCell(1);
                Cell cell2 = row.getCell(2);
                Cell cell3 = row.getCell(3);
                if (cell==null){
                    cell=row.createCell(0);
                }
                cell.setCellType(Cell.CELL_TYPE_STRING);  //设置第i行 第1格
                value0=cell.getStringCellValue();
                if (cell1==null){
                    cell1=row.createCell(1);
                }
                cell1.setCellType(Cell.CELL_TYPE_STRING);  //设置第i行 第1格
                value1=cell1.getStringCellValue();
                if (cell2==null){
                    cell2=row.createCell(2);
                }
                cell.setCellType(Cell.CELL_TYPE_STRING);  //设置第i行 第1格
                value2=cell2.getStringCellValue();
                if (cell3==null){
                    cell3=row.createCell(3);
                }
                cell.setCellType(Cell.CELL_TYPE_STRING);  //设置第i行 第1格
                value3=cell3.getStringCellValue();
                ArrayList<String> list = new ArrayList<String>();
                list.add(value0);
                list.add(value1);
                list.add(value2);
                list.add(value3);
                lists.add(list);
            }

           /* for (Row row : sheet) {
                ArrayList<String> list = new ArrayList<String>();
                for (Cell cell : row) {
                    //根据不同类型转化成字符串
                    cell.setCellType(Cell.CELL_TYPE_STRING);
                    list.add(cell.getStringCellValue());
                }

                lists.add(list);
            }*/
        } catch (IOException e) {
            e.printStackTrace();
        } catch (InvalidFormatException e) {

        } finally {
            try {
                if (inputStream != null) inputStream.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        return lists;
    }


    /**
     * 创建Excel.xls
     *
     * @param lists  需要写入xls的数据
     * @param titles 列标题
     * @param name   文件名
     * @return
     * @throws IOException
     */
    public static Workbook creatExcel(List<List<String>> lists, String[] titles, String name) throws
            IOException {
        System.out.println(lists);
        //创建新的工作薄
        Workbook wb = new HSSFWorkbook();
        // 创建第一个sheet（页），并命名
        Sheet sheet = wb.createSheet(name);
        // 手动设置列宽。第一个参数表示要为第几列设；，第二个参数表示列的宽度，n为列高的像素数。
        for (int i = 0; i < titles.length; i++) {
            sheet.setColumnWidth((short) i, (short) (35.7 * 150));
        }

        // 创建第一行
        Row row = sheet.createRow((short) 0);

        // 创建两种单元格格式
        CellStyle cs = wb.createCellStyle();
        CellStyle cs2 = wb.createCellStyle();

        // 创建两种字体
        Font f = wb.createFont();
        Font f2 = wb.createFont();

        // 创建第一种字体样式（用于列名）
        f.setFontHeightInPoints((short) 10);
        f.setColor(IndexedColors.BLACK.getIndex());
        f.setBoldweight(Font.BOLDWEIGHT_BOLD);

        // 创建第二种字体样式（用于值）
        f2.setFontHeightInPoints((short) 10);
        f2.setColor(IndexedColors.BLACK.getIndex());

        // 设置第一种单元格的样式（用于列名）
        cs.setFont(f);
        cs.setBorderLeft(CellStyle.BORDER_THIN);
        cs.setBorderRight(CellStyle.BORDER_THIN);
        cs.setBorderTop(CellStyle.BORDER_THIN);
        cs.setBorderBottom(CellStyle.BORDER_THIN);
        cs.setAlignment(CellStyle.ALIGN_CENTER);

        // 设置第二种单元格的样式（用于值）
        cs2.setFont(f2);
        cs2.setBorderLeft(CellStyle.BORDER_THIN);
        cs2.setBorderRight(CellStyle.BORDER_THIN);
        cs2.setBorderTop(CellStyle.BORDER_THIN);
        cs2.setBorderBottom(CellStyle.BORDER_THIN);
        cs2.setAlignment(CellStyle.ALIGN_CENTER);
        //设置列名
        for (int i = 0; i < titles.length; i++) {
            Cell cell = row.createCell(i);
            cell.setCellValue(titles[i]);
            cell.setCellStyle(cs);
        }
        if (lists == null || lists.size() == 0) {
            return wb;
        }
        //设置每行每列的值
        for (short i = 1; i <= lists.size(); i++) {
            // Row 行,Cell 方格 , Row 和 Cell 都是从0开始计数的
            // 创建一行，在页sheet上
            Row row1 = sheet.createRow((short) i);
            for (short j = 0; j < titles.length; j++) {
                // 在row行上创建一个方格
                Cell cell = row1.createCell(j);
                cell.setCellValue(lists.get(i - 1).get(j));
                cell.setCellStyle(cs2);
            }
        }
        return wb;
    }

    public static List<TaskSubFilter> filters = new ArrayList<>();

    public static void main(String[] args) {
        String path = "C:\\Users\\ling\\Desktop\\新建 Microsoft Excel 工作表 (2).xlsx";
        /* List<List<String>> lists = readExcel(path);
         *//* for (List<String> list : lists) {
            for (String strs : list) {
                System.out.println(strs);
            }
        }*//*
        List<Map<String, String>> mapList = new ArrayList<>();
        String s = "";
        for (int j = 1; j < lists.size(); j++) {
            List<String> list = lists.get(j);
            if (list.size() == 4) {
                Map<String, String> map = new HashMap<>();
                if (StringUtils.isBlank(list.get(0))) {
                    map.put("subName", s);
                } else {
                    s = list.get(0);
                    map.put("subName", s);
                }
                map.put("unit", list.get(1));
                map.put("unitType", list.get(2));
                map.put("timeLimit", list.get(3));
                mapList.add(map);
            }
        }
        List<String> list = new ArrayList<>();
        List<TaskUnit> unitList=new ArrayList<>();
        for (Map<String, String> map : mapList) {
            String subName = map.get("subName");
            list.add(subName);
            TaskUnit taskUnit = new TaskUnit();
            taskUnit.setDeptName(map.get("unit"));
            taskUnit.setTaskSubName(subName);
            taskUnit.setTimeLimit(map.get("timeLimit"));
            taskUnit.setUnitType(map.get("unitType"));
            unitList.add(taskUnit);
        }
        List<String> listNew=new ArrayList<>();
        for (String str:list) {
            if(!listNew.contains(str)){
                listNew.add(str);
            }
        }
        for (String s1 : listNew) {
            TaskSubFilter taskSubFilter = new TaskSubFilter();
            taskSubFilter.setSubName(s1);
            filters.add(taskSubFilter);
        }
        for (TaskSubFilter filter : filters) {
            List<TaskUnit>taskUnitList=new ArrayList<>();
            for (TaskUnit taskUnit : unitList) {
                if (taskUnit.getTaskSubName().equals(filter.getSubName())){
                    taskUnitList.add(taskUnit);
                }
            }
            filter.setTaskUnitList(taskUnitList);
        }
        System.out.println(filters);*/

    }
}